# -*- coding: utf-8 -*-
# @Time    : 2019/3/21 11:54
# @Author  : zjj
# @Email   : 1933860854@qq.com
# @File    : db_method.py
# @Software: PyCharm
import pymysql
host = '127.0.0.1'
port = 22799
password = '123456'
username = 'root'
dbName = 'test'
charset = 'utf8mb4'
cursorclass = pymysql.cursors.DictCursor

def db_query(query_sql):
    # 打开数据库连接
    db = pymysql.connect(host=host, port=port, user=username, passwd=password, db=dbName, charset=charset, cursorclass=cursorclass)
    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()
    # 使用 execute()  方法执行 SQL 查询
    cursor.execute(query_sql)
    # 使用 fetchone() 方法获取单条数据.
    data_list = cursor.fetchall()
    # 关闭数据库连接
    db.close()
    return data_list


def db_update(update_sql):
    # 打开数据库连接
    db = pymysql.connect(host=host, port=port, user=username, passwd=password, db=dbName, charset=charset)
    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()
    # 使用 execute()  方法执行 SQL 查询
    cursor.execute(update_sql)
    # 使用 fetchone() 方法获取单条数据.
    db.commit()
    # 关闭数据库连接
    db.close()

def db_update_params(update_sql, params):
    # 打开数据库连接
    db = pymysql.connect(host=host, port=port, user=username, passwd=password, db=dbName, charset=charset)
    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()
    # 使用 execute()  方法执行 SQL 查询
    cursor.execute(update_sql, params)
    # 使用 fetchone() 方法获取单条数据.
    db.commit()
    # 关闭数据库连接
    db.close()